from django import template

register = template.Library()


@register.filter
def first_active(num):
    '''
    第一个元素添加active样式
    :param num: int
    :return: string
    '''
    if num == 1:
        return 'active'
    return ''


@register.filter
def not_display_exclude_first(num):
    '''
    除第一个，加上display:none
    :param num:
    :return:
    '''
    if num == 1:
        return ''
    return 'display:none;'